package main

import (
	"fmt"
)

/*
定义一些初始值
username ="root"
passwd = "123456"
balance = 1000

实现一个简易的atm 存取款系统
1，登入成功才能进入系统
  选择：
   1，存钱
   2，取钱  -- 不能超过它的余额
   3，查询余额
   4，退出
2，登入失败有三次机会

*/
var  balance int   //声名全局变量初始余额
func main(){
	var(
		username string
		passwd string
		num1 int  //存放选择的数字
		money1 int  //需要存钱的数量
		money2 int //需要取的钱


	)
	balance = 1000  //账号初始余额
	reslut :=login1(username,passwd) //存放登入函数返回的值
	if reslut == true{
		fmt.Println("登入成功")
	}else {
		fmt.Println("登入失败")
	}
	for {   //一直循环，直到客户按4退出
		reslut3 := choice(num1)   //存放选择函数返回的值
		if reslut3 == 1 {  //取钱
			savemoney(money1)
		} else if reslut3 == 2 {  //存钱
			withdraw(money2)
		} else if reslut3 == 3 {  //查看余额
			balance1()
		} else if reslut3 == 4 {  //退出
			break
		}
	}

}
func login1(username,passwd string)bool {  //实现登入
	flag := 0
	for flag < 3 {
		fmt.Println("请输入用户名和密码：")
		fmt.Scan(&username,&passwd)
		if username == "root" && passwd == "123456" {
			return true
		}
		flag++
	}
	return false
}

func choice(num1 int) (ch int) {  //实现选择功能
	fmt.Println("欢迎登入三创银行")
	fmt.Println("按1存钱，按2取钱，按3查看余额，按4退出")
	fmt.Scan(&num1)
	return num1
}

func savemoney(money1 int)(bmoney int) {   //实现存钱功能
	//初始用户账号的钱
	fmt.Println("输入你要存钱的数量")
	fmt.Scan(&money1)
	balance += money1
	fmt.Println("你的余额为：",balance)
	return balance  //返回余额
}

func withdraw(money2 int) (num4 int) {   //实现取钱功能
	fmt.Println("你的余额为：",balance)
	fmt.Println("请输入你要取的金额")
	fmt.Scan(&money2)
	if money2 > balance{
		fmt.Println("余额不够")
	}else {
		balance -= money2
	}
	fmt.Println("你的余额为：",balance)
	return balance  //返回余额
}

func balance1(){   //实现查看余额功能
	fmt.Println("你的余额为：",balance)
}
